Python Centos系统下模型训练yolov5环境搭建

模型训练yolov5环境搭建

本文将介绍yolov5从环境搭建到模型训练的整个过程

yolov5下载

下载yolov5源码:https://github.com/ultralytics/yolov5
解压，可以看到里面有requirements.txt文件，里面记录了需要安装的包，这个txt文件可以帮助一键下载这些依赖包。

文件夹里也包含了train.py文件，这个也接下来训练yolo模型需要用到的启动文件。

接着上面的requirement.txt，介绍如何安装里面需要安装的依赖。首先打开下载好的yolov5_master 文件夹，在上面输入cmd回车，可以直接在该文件夹目录下打开命令行。

然后运行

pip install -r requirements.txt  -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

就会自动把这些依赖安装好了。

flask 安装

pip install flask -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

启动模型

python model_test_aqm.py

若没有报错则可正常启动

(base) [root@localhost yolov5-master]# python model_test_aqm.py 
 * Serving Flask app 'model_test_aqm'
 * Debug mode: off
WARNING: This is a development server. Do not use it in a production deployment. Use a production WSGI server instead.
 * Running on all addresses (0.0.0.0)
 * Running on http://127.0.0.1:5000
 * Running on http://192.168.23.134:5000
Press CTRL+C to quit

问题处理

1.出现这个错误 ModuleNotFoundError: No module named ‘cv2’

报错

 Traceback (most recent call last):
  File "/data/python/yolov5/yolov5-master/model_test_aqm.py", line 9, in <module>
    import cv2
ModuleNotFoundError: No module named 'cv2'

解决方法
安装这个两个库即可

pip install opencv-python-headless -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

pip install opencv-contrib-python -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

2.出现这个错误 libGL.so.1: cannot open shared object file: No such file or directory

报错

File "/usr/local/lib/python3.6/dist-packages/cv2/__init__.py", line 8, in <module>
    from .cv2 import *
ImportError: libGL.so.1: cannot open shared object file: No such file or directory

解决安装 ffmpeg libsm6 libxext6

安装libsm6 libxext6如下命令直接安装则可

yum install libSM-1.2.2-2.el7.x86_64 --setopt=protected_multilib=false

yum install libXrender-0.9.10-1.el7.x86_64 --setopt=protected_multilib=false

yum install libXext-1.3.3-3.el7.x86_64 --setopt=protected_multilib=false
``` 
安装ffmpeg

前提准备：使用yum安装编译工具 gcc cc cl

yum -y install gcc cc cl

1.安装ffmpeg时需要提前安装yasm插件

1)下载

wget http://www.tortall.net/projects/yasm/releases/yasm-1.3.0.tar.gz

2)解压

tar -xvf yasm-1.3.0.tar.gz

3)进入解压后的文件夹路径下，执行安装

cd yasm-1.3.0/

```
./configure && make && make install

2.安装FFmpeg

1)下载

wget http://www.ffmpeg.org/releases/ffmpeg-4.3.2.tar.gz

2)解压

tar -xvf ffmpeg-4.3.2.tar.gz

3)进入解压后的文件夹路径下，执行安装

cd ffmpeg-4.3.2/

./configure && make && make install

耗时相对长些…耐心等待….

3.查看安装结果

ffmpeg -version

出现如下类似结果，则安装成功

(yolo) [root@localhost ffmpeg-4.3.2]# ffmpeg -version
ffmpeg version 4.3.2 Copyright (c) 2000-2021 the FFmpeg developers
built with gcc 4.8.5 (GCC) 20150623 (Red Hat 4.8.5-44)
configuration: 
libavutil      56. 51.100 / 56. 51.100
libavcodec     58. 91.100 / 58. 91.100
libavformat    58. 45.100 / 58. 45.100
libavdevice    58. 10.100 / 58. 10.100
libavfilter     7. 85.100 /  7. 85.100
libswscale      5.  7.100 /  5.  7.100
libswresample   3.  7.100 /  3.  7.100

一辈子很短，努力的做好两件事就好；
第一件事是热爱生活，好好的去爱身边的人；
第二件事是努力学习，在工作中取得不一样的成绩，实现自己的价值，而不是仅仅为了赚钱；